tensor([[0.5100, 0.5100, 0.5100, 0.5100, 0.5100],
        [0.4900, 0.4900, 0.4900, 0.4900, 0.4900]], grad_fn=<RoundBackward1>)